from ultralytics import YOLO


def onnx():

    # load yolov8 segment model
    model = YOLO(".\\weight\\train\\breast_v6.pt")
    # Use the model
    success = model.export(format="onnx", opset=12, simplify=True)


if __name__ == "__main__":
    onnx()
